package com.dhcc.core.config;

import java.lang.reflect.Method;
import java.util.concurrent.Executor;
import java.util.concurrent.ThreadPoolExecutor;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.aop.interceptor.AsyncUncaughtExceptionHandler;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.AsyncConfigurer;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

import com.dhcc.core.config.properties.ThreadPoolProperties;  
 
/**
 * 默认线程池
 * @ClassName: AsyncTaskExecutePool 
 * @Description: 配置异步任务的线程池(注意：该线程池被所有的异步任务共享，而不属于某一个异步任务)
 * @author: cyf
 * @date: 2018年11月11日 下午5:06:50
 */
@Configuration
public class AsyncTaskExecutePool implements AsyncConfigurer{  
    private Logger logger = LoggerFactory.getLogger(getClass());
  
    @Autowired  
    private ThreadPoolProperties threadPoolProperties;  // 配置属性类，见上面的代码
  
	@Override
	public Executor getAsyncExecutor() {
		ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(threadPoolProperties.getCorePoolSize());  
        executor.setMaxPoolSize(threadPoolProperties.getMaxPoolSize());  
        executor.setQueueCapacity(threadPoolProperties.getQueueCapacity());  
        executor.setKeepAliveSeconds(threadPoolProperties.getKeepAliveSeconds()); 
        executor.setAllowCoreThreadTimeOut(threadPoolProperties.getAllowCoreThreadTimeout());
        executor.setThreadNamePrefix("medicalTaskExecutor-");  
  
        // rejection-policy：当pool已经达到max size的时候，如何处理新任务  
        // CALLER_RUNS：不在新线程中执行任务，而是由调用者所在的线程来执行  
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        executor.initialize();  
        return executor;  
	}
 
	@Override
	public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {// 异步任务中异常处理
		return new AsyncUncaughtExceptionHandler() {
			
			@Override
			public void handleUncaughtException(Throwable arg0, Method arg1, Object... arg2) {
			    logger.error("=========================="+arg0.getMessage()+"=======================", arg0);
				logger.error("exception method:"+arg1.getName());
			}
		};
	}  
}
